# 读 excel 文件
# openpyxl  读写 excel 文件

# 导入模块
import openpyxl
import ast
import yaml

# 打开 excel 文件
file = openpyxl.load_workbook('注册测试用例.xlsx')

# excel 文件具体读取的表格  通过下标来读取
data = file.worksheets[0]

# 定义一个空字典，存放测试数据
temp_data = {}

# temp_data['中国'] = 'China'

lines = 0
# 读取内容
for i in data.values:
    lines += 1
    if lines == 1:
        continue
    request_data = ast.literal_eval(i[4])
    request_data['excepts'] = int(i[6])
    temp_data[i[2]] = request_data
    print(i[2]) # 用例标题
    print(i[4]) # 请求报文
    print(i[6]) # 预期结果

# 写 yaml 中
# 写入yaml
f = open('./reg_data.yaml', 'w', encoding='utf-8')  # 打开文件
yaml.dump(temp_data, f, allow_unicode=True)  # 将数据写入
f.close()  # 关闭文件
